前言:
儲存上次進度訓練模組後,目前遇到一個小問題,載入方面一直沒試成功,所以我先做了一個辨別模型的if,以便於之後針對各種狀況切換訓練
程式碼:
if os.path.exists(DATASET_PATH):
if os.path.exists(DATASET_PATH + WEIGHTS_FINAL):
print(WEIGHTS_FINAL + "模型存在,將繼續訓練模型")
else:
print(WEIGHTS_FINAL + '模型不存在,將新建訓練模型')
# 訓練模型
net_final.fit(train_batches,
steps_per_epoch = train_batches.samples // BATCH_SIZE,
validation_data = valid_batches,
validation_steps = valid_batches.samples // BATCH_SIZE,
epochs = NUM_EPOCHS,
callbacks = [cp_callback])
else:
print(WEIGHTS_FINAL + '路徑不存在,請確認路徑')
# 儲存訓練好的模型
print("儲存訓練模型")
net_final.save(WEIGHTS_FINAL)